package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.heima.common.dto.NewsAuthDto;
import com.heima.common.dto.PageResponseResult;
import com.heima.common.dto.ResponseResult;
import com.heima.common.dto.User;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.util.MediaThreadLocalUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.dto.ImageDto;
import com.heima.media.dto.WmNewsDto;
import com.heima.media.dto.WmNewsPageDto;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmNewsMaterial;
import com.heima.media.entity.WmUser;
import com.heima.media.mapper.WmNewsMapper;
import com.heima.media.service.IAuditService;
import com.heima.media.service.IWmNewsMaterialService;
import com.heima.media.service.IWmNewsService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.heima.media.service.IWmUserService;
import com.heima.media.vo.WmNewsVo;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Lazy;
import org.springframework.kafka.core.KafkaTemplate;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * <p>
 * 自媒体图文内容信息表 服务实现类
 * </p>
 */
@Service
public class WmNewsServiceImpl extends ServiceImpl<WmNewsMapper, WmNews> implements IWmNewsService {

    @Override
    public ResponseResult listByCondition(WmNewsPageDto dto) {
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }
        // 查询条件
        LambdaQueryWrapper<WmNews> query = new LambdaQueryWrapper<>();
        // 当前登录用户只能查看自己发布的文章
        query.eq(WmNews::getUserId, user.getUserId());
        // 如果状态不为空,添加状态查询
        if (dto.getStatus() != null) {
            query.eq(WmNews::getStatus, dto.getStatus());
        }
        // 如果关键字不为空,根据文章标题模糊查询
        if (!StringUtils.isEmpty(dto.getKeyword())) {
            query.like(WmNews::getTitle, dto.getKeyword());
        }
        // 如果频道id不为空,根据频道id查询
        if (dto.getChannelId() != null) {
            query.eq(WmNews::getChannelId, dto.getChannelId());
        }
        // 如果开始时间不为空,查询发布日期大于等于开始时间
        if (dto.getBeginPubDate() != null) {
            query.ge(WmNews::getPublishTime, dto.getBeginPubDate());
        }
        // 如果结束时间不为空,查询发布日期小于开始时间
        if (dto.getEndPubDate() != null) {
            query.lt(WmNews::getPublishTime, dto.getEndPubDate());
        }
        // 根据发布时间倒序排
        query.orderByDesc(WmNews::getPublishTime);
        // 分页查询
        IPage<WmNews> page = new Page<>(dto.getPage(), dto.getSize());
        IPage<WmNews> iPage = this.page(page, query);
        PageResponseResult result = new PageResponseResult(dto.getPage(), dto.getSize(), iPage.getTotal(), iPage.getRecords());
        return result;
    }

    @Autowired
    @Lazy   // 懒加载
    private IAuditService auditService;

    @Override
    public ResponseResult submit(WmNewsDto dto) {
        User user = MediaThreadLocalUtil.get();
        if (user == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NEED_LOGIN);
        }
        // 保存无图类型的文章

        // 判断如果dto中有文章id,说明是需要更新,需要将原有的关联关系删除
        if (dto.getId() != null) {
            removeRelation(dto.getId());
        }

        // 构建 WmNews 对象,保存到数据库中
        WmNews wmNews = new WmNews();
        BeanUtils.copyProperties(dto, wmNews);
        wmNews.setUserId(user.getUserId());
        wmNews.setCreatedTime(new Date());
        wmNews.setSubmitedTime(new Date());
        wmNews.setEnable(true);
        // 将封面图片转成json存储在 wm_news 中
        wmNews.setImages(JSON.toJSONString(dto.getImages()));

        // 从内容中提取图片
        List<ImageDto> contentImages = getImageFromContent(dto.getContent());
        // 如果前端传递是布局是 -1 自动,需要从内容中提取封面图片
        if (dto.getType() == -1) {
            if (contentImages.size() <= 0) {
                // 内容中无图,设置布局为无图 type = 0
                wmNews.setType(0);
            } else if (contentImages.size() < 3) {
                // 内容中图片小于3张,设置布局为单图,type = 1
                wmNews.setType(1);
                // 提取内容中的第一张图片作为封面图片
                wmNews.setImages(JSON.toJSONString(contentImages.subList(0, 1)));
            } else {
                // 内容中图片大于等于3张,设置布局为多图,type = 3
                wmNews.setType(3);
                // 提取内容中的前三张图片作为封面图片
                wmNews.setImages(JSON.toJSONString(contentImages.subList(0, 3)));
            }
        }

        // this.save(wmNews);
        this.saveOrUpdate(wmNews);  // 判断有没有id,有id执行update ,没有id 执行save


        // 如果状态不是草稿,是提交审核,修改保存素材和文章的关系
        if (dto.getStatus() != 0) {
            // 保存封面图片和文章的关联关系
            // 封面图片从对象中重新解析出来
            List<ImageDto> coverImages = JSON.parseArray(wmNews.getImages(), ImageDto.class);
            saveRelation(coverImages, wmNews, 1);
            // saveRelation(dto.getImages(), wmNews, 1);
            // 保存内容图片和文章的关联关系
            saveRelation(contentImages, wmNews, 0);

            // 开启自动审核流程
            System.out.println(Thread.currentThread().getName() + "调用审核服务...");
            auditService.audit(wmNews);
        }

        return ResponseResult.okResult();
    }

    @Autowired
    private KafkaTemplate<String, String> kafkaTemplate;

    @Override
    public ResponseResult enable(WmNewsDto dto) {
        // 根据id查询文章
        WmNews wmNews = this.getById(dto.getId());
        // 判断文章是否存在
        if (wmNews == null) {
            return ResponseResult.errorResult(AppHttpCodeEnum.DATA_NOT_EXIST);
        }
        // 判断文章是否已发布
        if (wmNews.getStatus() != 9) {
            return ResponseResult.errorResult(AppHttpCodeEnum.NO_OPERATOR_AUTH);
        }
        // 更新上下架状态
        wmNews.setEnable(dto.getEnable() == 1 ? true : false);
        this.updateById(wmNews);
        // 将上下架状态同步到article端

        // 将文章库id作为key
        // 将状态作为value
        String key = wmNews.getArticleId() + "";
        String value = dto.getEnable() + "";
        kafkaTemplate.send(enableTopic, key, value);

        return ResponseResult.okResult();
    }

    @Autowired
    private IWmUserService userService;

    @Override
    public ResponseResult findPageByName(NewsAuthDto dto) {
        // 构建分页条件
        IPage<WmNews> page = new Page<>(dto.getPage(), dto.getSize());
        // 构建查询条件
        LambdaQueryWrapper<WmNews> query = new LambdaQueryWrapper<>();
        // 根据标题模糊查询
        if (!StringUtils.isEmpty(dto.getTitle())) {
            query.like(WmNews::getTitle, dto.getTitle());
        }
        // 根据状态查询
        if (dto.getStatus() != null) {
            query.eq(WmNews::getStatus, dto.getStatus());
        }
        // 根据发布时间倒序排列
        query.orderByDesc(WmNews::getPublishTime);
        // 分页查询
        IPage<WmNews> iPage = this.page(page, query);
        List<WmNews> newsList = iPage.getRecords();

        // 构建vo对象
        List<WmNewsVo> vos = new ArrayList<>();
        for (WmNews wmNews : newsList) {
            WmNewsVo vo = new WmNewsVo();
            // 属性赋值
            BeanUtils.copyProperties(wmNews, vo);
            // 查询作者信息
            // todo 优化思路: 获取当前页数据所有的用户id,做一次in查询,拿到结果后对应userId设置name
            WmUser wmUser = userService.getById(wmNews.getUserId());
            vo.setAuthorName(wmUser.getName());
            // 加入到集合中
            vos.add(vo);
        }
        // 返回通用的分页响应
        PageResponseResult result = new PageResponseResult(dto.getPage(), dto.getSize(),
                iPage.getTotal(), vos);
        return result;
    }

    @Override
    public ResponseResult<WmNewsVo> findNewsVoById(Integer id) {
        // 根据id查询文章
        WmNews wmNews = this.getById(id);
        // 查询作者名称
        WmUser wmUser = userService.getById(wmNews.getUserId());
        // 构建vo对象
        WmNewsVo vo = new WmNewsVo();
        BeanUtils.copyProperties(wmNews, vo);
        vo.setAuthorName(wmUser.getName());
        // 通用响应
        return ResponseResult.okResult(vo);
    }

    @Override
    public ResponseResult auth(NewsAuthDto dto, int type) {
        // 查询文章
        WmNews wmNews = this.getById(dto.getId());
        // 审核不通过 修改状态为2 记录失败原因
        if (type == 0) {
            wmNews.setStatus(2);
            wmNews.setReason(dto.getMsg());
            this.updateById(wmNews);
        } else {
            // 将状态改为4
            wmNews.setStatus(4);
            this.updateById(wmNews);
            // 审核通过 依旧走自动审核的逻辑
            auditService.audit(wmNews);
        }
        return ResponseResult.okResult();
    }

    @Value("${topic.enableTopic}")
    private String enableTopic;

    /**
     * 根据文章id删除关联记录
     *
     * @param newsId
     */
    private void removeRelation(Integer newsId) {
        LambdaQueryWrapper<WmNewsMaterial> query = new LambdaQueryWrapper<>();
        query.eq(WmNewsMaterial::getNewsId, newsId);
        newsMaterialService.remove(query);
    }

    /**
     * 从内容中提取图片
     *
     * @param content
     * @return
     */
    private List<ImageDto> getImageFromContent(String content) {
        List<ImageDto> list = new ArrayList<>();
        // 将json转换成对象集合
        List<ContentDto> dtos = JSON.parseArray(content, ContentDto.class);
        // 遍历,查找type = image
        for (ContentDto dto : dtos) {
            if (dto.getType().equals("image")) {
                ImageDto imageDto = new ImageDto();
                imageDto.setId(dto.getId());
                imageDto.setUrl(dto.getValue());
                list.add(imageDto);
            }
        }
        return list;
    }

    @Autowired
    private IWmNewsMaterialService newsMaterialService;

    /**
     * 保存素材和文章关联关系
     *
     * @param images
     * @param wmNews
     * @param type
     */
    private void saveRelation(List<ImageDto> images, WmNews wmNews, int type) {
        int ord = 0;
        // 遍历图片
        for (ImageDto image : images) {
            // 构建关联对象
            WmNewsMaterial newsMaterial = new WmNewsMaterial();
            newsMaterial.setMaterialId(image.getId());
            newsMaterial.setNewsId(wmNews.getId());
            newsMaterial.setType(type);
            newsMaterial.setOrd(ord);
            ord++;
            // 保存关联对象
            newsMaterialService.save(newsMaterial);
        }

    }
}
